#pragma once

#include <stdlib.h>
#include <cstdio>
#include <iostream>
#include <unistd.h>
#include <signal.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#define root "./"
#define devnull "/dev/null"

void Daemon(bool ischdir, bool isclose = false)
{
    // * 1.守护进程一般要屏蔽常见的信号
    signal(SIGCHLD, SIG_IGN);
    signal(SIGPIPE, SIG_IGN);

    // * 2. 成为非组长
    if (fork() > 0)
        exit(0);

    // * 3. 建立会话
    setsid();
    // * 4.每一个进程都有自己的 cwd， 现在要求询问是否需要改变自己的cwd
    if (ischdir)
        chdir(root);

    // * 5. 已经百年未守护进程， 不需要用户进行输入和输出， 错误进行关联了
    if (isclose)
    {
        ::close(0);
        ::close(1);
        ::close(2);
    }
    else
    {
        // 各种重定向
        int fd = ::open(devnull, O_WRONLY);
        if (fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            close(fd);
        }
    }
}